import torch
from transformers.utils import is_torch_greater_or_equal
print(is_torch_greater_or_equal("2.5"))
print(torch.distributed.is_available())
print("PyTorch版本:", torch.__version__)
print("分布式可用:", torch.distributed.is_available())
print("CUDA可用:", torch.cuda.is_available())
#pip install torch>=2.5.0 torchvision>=0.15.0 torchaudio>=2.5.0 --index-url https://download.pytorch.org/whl/cu121